#include "BlockQueue.hpp"
#include "Task.hpp"
#include <pthread.h>
#include <ctime>
#include <sys/types.h>
#include <unistd.h>

class ThreadData
{
public:
    BlockQueue<Task> *bq;
    std::string name;
};

void *consumer(void *args)
{
    ThreadData *td = (ThreadData*)args;
    // BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    while (true)
    {
        // sleep(1);
        Task t;
        // 1. 消费数据 bq->pop(&data);
        td->bq->Pop(&t);
       
        // 2. 进行处理
        // t.Run();
        t(); // 处理任务的时候，会不会消耗时间呢？？？
        std::cout << "consumer data: " << t.PrintResult() << ", " << td->name << std::endl;
        // 注意：消费者没有sleep！！！
    }

    return nullptr;
}

void *productor(void *args)
{
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    // BlockQueue<int> *bq = static_cast<BlockQueue<int> *>(args);
    // BlockQueue *bq = static_cast<BlockQueue *>(args); // 生产者和消费者同时看到了一份公共资源
    while (true)
    {
        // 1. 数据可以从具体场景中来，从网络中拿数据
        int data1 = rand() % 10;
        usleep(rand()%123);
        int data2 = rand() % 10;
        usleep(rand()%123);
        char oper = opers[rand() % (opers.size())];
        Task t(data1, data2, oper);
        std::cout << "productor task: " << t.PrintTask() << std::endl;
        // 2. 进行生产
        // bq->Push(data);
        bq->Push(t);
        sleep(1);

        // for debug

    }

    return nullptr;
}

int main()
{
    srand((uint16_t)time(nullptr) ^ getpid() ^ pthread_self()); // 只是为了形成更随机的数据
    // 既然是模板，就不止是可以传递int/double等，还可以让生产者给消费者分派任务
    BlockQueue<Task> *bq = new BlockQueue<Task>();
    pthread_t c[3], p[2]; // 消费者和生产者：支持多生产者和多消费者，也就是多线程

    ThreadData *td = new ThreadData();
    td->bq = bq;
    td->name = "thread-1";
    pthread_create(&c[0], nullptr, consumer, td);


    ThreadData *td1 = new ThreadData();
    td1->bq = bq;
    td1->name = "thread-2";
    pthread_create(&c[0], nullptr, consumer, td1);

    ThreadData *td2 = new ThreadData();
    td2->bq = bq;
    td2->name = "thread-3";
    pthread_create(&c[0], nullptr, consumer, td2);


    pthread_create(&p[0], nullptr, productor, bq);
    pthread_create(&p[1], nullptr, productor, bq);

    pthread_join(c[0], nullptr);
    pthread_join(c[1], nullptr);
    pthread_join(c[2], nullptr);
    pthread_join(p[0], nullptr);
    pthread_join(p[1], nullptr);

    return 0;
}
